from ascend_yolo_model import AscendYOLOModel

if __name__ == '__main__':
    ascend_model = AscendYOLOModel(model_path='/app/dependences/Yolov8_for_PyTorch/model/yolov8n-seg.pt',
                                   task='segment',
                                   om_model_path='/app/dependences/Yolov8_for_PyTorch/model/yolov8n-seg_bs1.om',
                                   device_id=0)
    print(ascend_model.predict(source='/app/dependences/Yolov8_for_PyTorch/person.jpg', save=True))
